package com.hibikiluler.javalearning.util;

import io.jsonwebtoken.Claims;
import io.jsonwebtoken.Jwt;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.security.core.userdetails.UserDetails;
import org.springframework.stereotype.Component;

import java.util.Date;
import java.util.HashMap;
import java.util.Map;

/**
 * Token 工具类 生成token、
 * 用户拿到token 访问资源
 *     #请求头
 *     tokenheader: Authorization
 *     #签名
 *     secret : hibikiluler
 *     # jwt 过期时间
 *     expiration: 3600
 *     #token头部
 *     tokenhead: "Bearer "
 */
@Component
public class TokenUtils {

    @Value("${jwt.secret}")  //注入xml里面的secret
    private String secret;

    @Value("${jwt.expiration}")  //注入xml里面的secret
    private long expiration;

    //传入用户登录信息，生成token
    public String generateToken(UserDetails userDetails){
        Map<String,Object> map = new HashMap<>(2);
        map.put("userName",userDetails.getUsername());
        map.put("created", new Date());
        return this.generateJwt(map);
    }
    /**
     * 根据荷载信息生成token
     */
    private String generateJwt(Map<String,Object> map){
        return Jwts.builder()
                .setClaims(map)
                .signWith(SignatureAlgorithm.HS512,secret)
                .setExpiration(new Date(System.currentTimeMillis() + expiration * 1000))
                .compact();
    }

    /**
     * 根据token 获取荷载信息
     * @param token
     * @return
     */
    public Claims getTokenBody(String token){
        try {
            return Jwts.parser()
                    .setSigningKey(secret)
                    .parseClaimsJws(token)
                    .getBody();
        } catch (Exception e){
            return null;
        }
    }

    /**
     * 根据token得到用户名
     * @param token
     * @return
     */
    public String getUsernameByToken(String token){
        return (String) this.getTokenBody(token).get("userName");
    }

    // 判断token是否过期
    public boolean isExpiration(String token){
        return getTokenBody(token).getExpiration().before(new Date());
    }

    //刷新token
    public String Refresh(String token){
        Claims claims = this.getTokenBody(token);
        claims.setExpiration(new Date());
        return this.generateJwt(claims);
    }
}

